package algorithm7.leetcode.hot100.c1哈希;

import org.junit.jupiter.api.Test;

import java.util.*;

/**
 * @author merrick.peng
 */
public class h2_LC49字母异位词分组 {

    public List<List<String>> groupAnagrams(String[] strs) {
//        return handle1(strs);
        return handle2(strs);
    }

    private List<List<String>> handle2(String[] strs) {
        Map<String, List<String>> map = new HashMap<>();
        for (String str : strs) {
            char[] chars = str.toCharArray();
            Arrays.sort(chars);
            String key = Arrays.toString(chars);
            List<String> groupThis = map.computeIfAbsent(key, k -> new ArrayList<>());
            groupThis.add(str);
        }
        return new ArrayList<>(map.values());
    }


    private List<List<String>> handle1(String[] strs) {
        Map<String, List<String>> map = new HashMap<>();
        for (String str : strs) {
            char[] charArray = str.toCharArray();
            Arrays.sort(charArray);
            String key = Arrays.toString(charArray);
            List<String> value = map.getOrDefault(key, new ArrayList<>());
            value.add(str);
            map.put(key, value);
        }
        return new ArrayList<>(map.values());
    }

    /*

    给你一个字符串数组，请你将 字母异位词 组合在一起。可以按任意顺序返回结果列表。
    字母异位词 是由重新排列源单词的所有字母得到的一个新单词。

    示例 1:
    输入: strs = ["eat", "tea", "tan", "ate", "nat", "bat"]
    输出: [["bat"],["nat","tan"],["ate","eat","tea"]]

    示例 2:
    输入: strs = [""]
    输出: [[""]]

    示例 3:
    输入: strs = ["a"]
    输出: [["a"]]

    */

    @Test
    void test() {
        String[] strs = {"eat", "tea", "tan", "ate", "nat", "bat"};
        List<List<String>> res = groupAnagrams(strs);
        for (List<String> list : res) {
            System.out.println(list);
        }
    }

}
